#pragma once
#include <map>
#include "ProbabilityModel.h"

class MockProbabilityModel : public ProbabilityModel
{
	public:
		MockProbabilityModel(const char* aWord, int wordLength, unsigned int symbolSize = 8);
		virtual ~MockProbabilityModel(void);

		virtual int getUpperBoundForSymbol(unsigned int symbol);
		virtual int getLowerBoundForSymbol(unsigned int symbol);
		virtual int getAllSymbolsCount();
		virtual unsigned int getSymbolFromCodeNumber(unsigned int codeNumber);
		virtual unsigned int getSymbolSize();

	private:
		std::map<unsigned int, unsigned int> symbolsCount;
		const char* word;
		unsigned int symbolSize;
		int count;
};

